How do I create a multidimensional array in Python similar to a Java example